const cors = require('@koa/cors');
const config = require('../../config/index');

function register(app) {
  const domains = [];
  Object.keys(config).forEach(key => {
    domains.push(`http://${config[key].host}:${config[key].PORT}`);
  });

  app.use(
    cors({
      credentials: true,
      origin: '*',
      // origin(ctx) {
      //   if (domains.indexOf(ctx.request.header.origin) !== -1) {
      //     return ctx.request.header.origin;
      //   }
      //   return domains[0];
      // },
    }),
  );
}

module.exports = {
  register,
};
